package leetcode;

import util.TreeNode;

import java.util.ArrayList;
import java.util.List;

/**
 * @Project : algorithm-learning-java
 * @Author : Ruoyu Wang
 * @User : Momenta
 * @DateTime : 2024/1/26 17:06
 */
//public class Q0103 {
//    List<List<Integer>> res = new ArrayList<>();
//
//    private void dfs(TreeNode node, int k) {
//        if (node == null) return;
//        if (k > res.size()) res.add(new ArrayList<>());
//        if (k % 2 == 1) res.get(k - 1).add(node.val);
//        else res.get(k - 1).add(0, node.val);
//        dfs(node.left, k + 1);
//        dfs(node.right, k + 1);
//    }
//
//    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
//        dfs(root, 1);
//        return res;
//    }
//}


public class Q0103 {
    List<List<Integer>> res = new ArrayList<>();

    private void dfs(TreeNode node, int k) {
        if (node == null) return;
        if (k >= res.size()) res.add(new ArrayList<>());
        if (k % 2 == 1) res.get(k).add(0, node.val);
        else res.get(k).add(node.val);
        dfs(node.left,k+1);
        dfs(node.right,k+1);
    }

    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        dfs(root, 0);
        return res;
    }
}